EfficientDetLite0

object EfficientDetLite0 : ONNXModels.ObjectDetection<SSDLikeModel>

This model is a real-time neural network for object detection that detects 90 different classes (labels are available in org.jetbrains.kotlinx.dl.impl.dataset.Coco.V2017).

Internally it uses the EfficientNetLite as backbone network.

The model have an input with the shape is (1x320x320x3).

The model has 4 outputs:

  • num_detections: the number of detections.

  • detection_boxes: a list of bounding boxes. Each list item describes a box with top, left, bottom, right relative to the image size.

  • detection_scores: the score for each detection with values between 0 and 1 representing probability that a class was detected.

  • detection_classes: Array of 10 integers (floating point values) indicating the index of a class label from the COCO class.

NOTE: The detections are limited to 25.

See also

Functions

model
Link copied to clipboard
open fun model(modelHub: ModelHub): OnnxInferenceModel
pretrainedModel
Link copied to clipboard
open override fun pretrainedModel(modelHub: ModelHub): SSDLikeModel

Properties

inputShape
Link copied to clipboard
open val inputShape: LongArray?

Shape of the input accepted by this model, without batch size.

modelRelativePath
Link copied to clipboard
open override val modelRelativePath: String
preprocessor
Link copied to clipboard
open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>